package dictController

import (
	"CMETO/models"
	"CMETO/utils"
	"net/http"
	"strconv"
	"time"

	"github.com/gin-gonic/gin"
)

type Dictcontroller struct{}

// @Summary 字典查询
// @Description 查询字典（分页）
// @Tags 字典管理
// @Accept json
// @Produce json
// @Param page query string false "页码"
// @Param pageSize query string false "尺寸"
// @Param name query string false "字典名"
// @Success 200 {object} models.ResultCommon[[]models.Dict] "查询成功"
// @Success 400 {object} models.ErrorResult "查询失败"
// @Router /dict [get]
// @Security JWT
func (dict Dictcontroller) QueryDicts(ctx *gin.Context) {
	// 获取分页参数
	page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(ctx.DefaultQuery("pageSize", "10"))
	nameStr := ctx.DefaultQuery("name", "")
	name := "%" + nameStr + "%"
	dictList := []models.Dict{}
	models.DB.Model(&models.Dict{}).Where("is_delete!=1 AND name like ?", name).Limit(pageSize).Offset(pageSize * (page - 1)).Preload("DictValue").Find(&dictList)
	var total int64
	// 查询条目
	models.DB.Table("dicts").Where("is_delete!=1 AND name like ?", name).Count(&total)
	// 返回值约束
	result := &models.ResultByPage[models.Dict]{
		Msg:      "ok",
		Code:     "200",
		Total:    total,
		Page:     page,
		PageSize: pageSize,
		Result:   dictList,
	}
	ctx.JSON(http.StatusOK, result)
}

// @Summary 字典新增
// @Description 新增字典
// @Tags 字典管理
// @Accept json
// @Produce json
// @Param deptInfo body models.Dict true "字典信息"
// @Success 200 {object} models.AddResult "新增成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.AddResult "新增失败"
// @Router /dict [post]
// @Security JWT
func (d Dictcontroller) AddDict(ctx *gin.Context) {
	dict := &models.Dict{}
	// 绑定json数据
	if err := ctx.ShouldBindJSON(&dict); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "数据错误",
			Code:   "400",
			Result: nil,
		})
	} else {
		dict.CreateTime = utils.TimeFormatter(time.Now())
		var result = models.DB.Model(&models.Dict{}).Create(&dict).RowsAffected
		// 返回
		if result != 1 {
			ctx.JSON(http.StatusOK, &models.AddResult{
				Msg:    "插入失败",
				Code:   "400",
				Result: result,
			})
		} else {
			ctx.JSON(http.StatusOK, &models.AddResult{
				Msg:    "插入成功",
				Code:   "200",
				Result: result,
			})
		}
	}
}

// @Summary 字典修改
// @Description 修改字典
// @Tags 字典管理
// @Accept json
// @Produce json
// @Param deptInfo body models.Dict true "字典信息"
// @Success 200 {object} models.UpdateResult[models.Dict] "更新成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.AddResult "修改失败"
// @Router /dict [put]
// @Security JWT
func (d Dictcontroller) UpdateDict(ctx *gin.Context) {
	var dict models.Dict
	// 绑定json数据
	if err := ctx.ShouldBindJSON(&dict); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "数据绑定失败",
			Code:   "400",
			Result: err,
		})
		return
	}
	dict_old := &models.Dict{}
	// 开启事务
	tx := models.DB.Begin()
	// 捕获异常，异常回滚
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
			ctx.JSON(http.StatusOK, &models.ErrorResult{
				Msg:    "更新失败",
				Code:   "400",
				Result: nil,
			})
		}
	}()
	models.DB.First(&dict_old, dict.Id)
	if dict_old.Id == 0 {
		panic("字典不存在")
	}
	dictValue := &models.DictValue{}
	tx.Model(&dict_old).Updates(dict)
	tx.Where("dict_id = ?", dict_old.Id).Delete(&dictValue)
	result := tx.Model(&dictValue).Create(dict.DictValue).RowsAffected
	if result == 0 {
		tx.Rollback()
		panic("更新失败")
	} else {
		tx.Commit()
		ctx.JSON(http.StatusOK, &models.UpdateResult[models.Dict]{
			Msg:    "更新成功",
			Code:   "200",
			Result: dict,
		})
	}
}

// @Summary 删除字典
// @Description 传入json，批量删除字典
// @Tags 字典管理
// @Accept json
// @Produce json
// @Param deptsIds body models.DeleteParams true "字典id合集"
// @Success 200 {object} models.DeleteResult "更新成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.DeleteResult "删除失败"
// @Router /dict [delete]
// @Security JWT
func (d Dictcontroller) DeleteDict(ctx *gin.Context) {
	// 绑定参数实例
	ids := &models.DeleteParams{}
	// 定义模型实例
	var dict models.Dict
	if err := ctx.ShouldBindJSON(&ids); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "删除失败",
			Code:   "400",
			Result: err,
		})
	} else {
		// 假的删
		result := models.DB.Model(&dict).Where("id IN (?)", ids.Ids).Update("is_delete", 1)
		if result.RowsAffected != 0 {
			ctx.JSON(http.StatusOK, &models.DeleteResult{
				Msg:    "删除成功",
				Code:   "200",
				Result: ids.Ids,
			})
		} else {
			ctx.JSON(http.StatusOK, &models.DeleteResult{
				Msg:    "删除失败",
				Code:   "400",
				Result: []int{0},
			})
		}
	}
}
